package com.ybkj.daijia.server.websocket;

import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import org.springframework.util.Assert;

public class WebSocketSessions {

    private static final Map<String, WebSocketSessionEx> map = Collections
        .synchronizedMap(new HashMap<String, WebSocketSessionEx>());

    public static void addSession(String userId, WebSocketSessionEx session) {

        Assert.notNull(userId);
        Assert.notNull(session);

        map.put(userId, session);
    }

    public static WebSocketSessionEx getSession(String userId) {

        return map.get(userId);
    }

    public static Collection<WebSocketSessionEx> getAllSession() {

        return map.values();
    }

    public static void removeSession(String userId) {
        map.remove(userId);
    }
}
